#include <iostream>
#include "keyword_processor.hpp"
#include "page_processor.hpp"


int main(int argc, char* argv[]) {
    if (argc != 4) {
        std::cerr << "用法: " << argv[0] << " <中文语料目录> <英文语料目录> <网页语料目录>" << std::endl;
        return 1;
    }
    // 第一部分：关键字推荐（生成词典库和索引库）
    std::cout << "===== 开始处理关键字推荐模块 =====" << std::endl;
    KeyWordProcessor kwProcessor;
    kwProcessor.process(argv[1], argv[2]);
    std::cout << "===== 关键字推荐模块处理完成 =====" << std::endl;
    // 第二部分：网页搜索（生成网页库、偏移库和倒排索引）
    std::cout << "\n===== 开始处理网页搜索模块 =====" << std::endl;
    PageProcessor pageProcessor;
    pageProcessor.process(argv[3]);
    std::cout << "===== 网页搜索模块处理完成 =====" << std::endl;
    return 0;
}